encoder_name="conformer_cat" # conformer_cat | ecapa_tdnn_large
embedding_dim=192
loss_name="amsoftmax"

dataset="vox"
num_classes=7205
num_blocks=6
train_csv_path="data/train.csv"

input_layer=conv2d2
pos_enc_layer_type=rel_pos # no_pos| rel_pos | abs_pos
save_dir=test

checkpoint_path1=/apdcephfs/share_1157259/users/autozhang/con_exp4/conv2d2/ecapa_tdnn_large_6_192_amsoftmax/epoch=13_cosine_eer=1.47.ckpt
checkpoint_path2=/apdcephfs/share_1157259/users/autozhang/con_exp4/conv2d2/ecapa_tdnn_large_6_192_amsoftmax/epoch=13_cosine_eer=1.47.ckpt

checkpoint_path1=/apdcephfs/share_1157259/users/autozhang/con_exp4/conv2d/se_resnet34_6_192_amsoftmax/epoch=4_cosine_eer=1.63.ckpt
checkpoint_path2=/apdcephfs/share_1157259/users/autozhang/con_exp4/conv2d/se_resnet34_6_192_amsoftmax/epoch=4_cosine_eer=1.63.ckpt

#checkpoint_path1=/apdcephfs/share_1157259/users/autozhang/con_exp4/conv2d/ecapa_tdnn_3_192_amsoftmax/epoch=14_cosine_eer=1.10.ckpt
#checkpoint_path2=/apdcephfs/share_1157259/users/autozhang/con_exp4/conv2d/ecapa_tdnn_3_192_amsoftmax/epoch=14_cosine_eer=1.10.ckpt

#checkpoint_path1=/apdcephfs/share_1157259/users/autozhang/con_exp4/conv2d/ecapa_tdnn_large_3_192_amsoftmax/epoch=22_cosine_eer=1.04.ckpt
#checkpoint_path2=/apdcephfs/share_1157259/users/autozhang/con_exp4/conv2d/ecapa_tdnn_large_3_192_amsoftmax/epoch=22_cosine_eer=1.04.ckpt

checkpoint_path1=/apdcephfs/share_1157259/users/autozhang/con_exp2/conv2d2/conformer_cat_6_192_amsoftmax/epoch=13_cosine_eer=1.48.ckpt
checkpoint_path2=/apdcephfs/share_1157259/users/autozhang/con_exp2/conv2d2/conformer_cat_6_192_amsoftmax/epoch=15_cosine_eer=1.48.ckpt


python3 eval_test.py \
	--embedding_dim $embedding_dim \
	--save_dir $save_dir \
	--encoder_name $encoder_name \
	--train_csv_path $train_csv_path \
	--learning_rate 0.001 \
	--encoder_name ${encoder_name} \
	--num_classes $num_classes \
	--loss_name $loss_name \
    --num_blocks $num_blocks \
    --step_size 4 \
    --gamma 0.5 \
    --input_layer $input_layer \
    --pos_enc_layer_type $pos_enc_layer_type \
    --cohort_path data/cohort.csv \
    --top_n 128 \
    --norm \
	--trial_path /apdcephfs/share_1157259/users/autozhang/sitw/eval/sitw_eval.list \
    --checkpoint_path1 $checkpoint_path1 \
    --checkpoint_path2 $checkpoint_path2

	#--trial_path data/vox1_test.txt \
	#--trial_path /apdcephfs/share_1157259/users/autozhang/sitw/dev/sitw_dev.list \
	#--trial_path /apdcephfs/share_1157259/users/autozhang/sitw/eval/sitw_eval.list \
rm -rf test
